package project.base.common.database

/**
 * 数据源上下文
 * @author: 三味
 * @update: 2022/3/2 20:16
 */
object DbContextHolder {
    /**
     * 动态数据源名称上下文
     */
    private val dataSources = ThreadLocal<String>()

    /**
     * 设置/切换数据源
     */
    fun setDataSource(customerType: String) {
        dataSources.set(customerType)
    }

    /**
     * 获取数据源名称
     */
    fun getDataSource(): String? {
        return dataSources.get()
    }

    /**
     * 删除当前数据源名称
     */
    fun clearDataSource() {
        dataSources.remove()
    }
}
